'''chatglm3 调用的函数'''
tools = [
    {
        "name": "search_water_account",
        "description": "查询户号",
        "parameters": {
            "type": "object",
            "properties": {
                "address": {
                    "type": "string",
                    "description": "账号地址"
                }
            },
            "required": ["address"]
        }
    },
    {
        "name": "search_water_bill",
        "description": "查询水费账单",
        "parameters": {
            "type": "object",
            "properties": {
                "account_number": {
                    "type": "string",
                    "description": "用户的户号"
                },
                "month": {
                    "type": "string",
                    "description": "水费账单的月份"
                }
            },
            "required": ["account_number","month"]
        }
    },
    {
        "name": "search_water_notice",
        "description": "查询停水公告",
        "parameters": {
            "type": "object",
            "properties": {
                "address": {
                    "type": "string",
                    "description": "停水地址"
                }
            },
            "required": ["address"]
        }
    },
    {
        "name": "query_fee_standards",
        "description": "查询收费标准",
        "parameters": {
            "type": "object",
            "properties": {
                "address": {
                    "type": "string",
                    "description": "用水类型"
                }
            },
            "required": ["address"]
        }
    }
]

#水厂客服人员角色扮演
system_info = {"role": "system", "content": "作为水厂电话客服工作人员，你的主要职责是回答客户咨询的问题。\
        你需要解决客户的查询停水公告, 查询水费账单、查询收费标准, 故障报修和问题咨询等。请提供详细和专业的回答。\
        注意回答要清晰、准确，并以专业方式回答。同时，回答应涵盖所有客户咨询的类型，并尽可能提供详细的解决方案和建议。\
        确保回答能够令客户满意并解决问题。让我们一步一步来思考 。\
        对于用户提出的问题, 如果是查询停水公告、查询户号、查询水费账单、查询收费标准, 你可以访问以下工具：", "tools": tools}

#查询户号
def search_water_account(params):
    return "户号:1234567"

#查询水费账单
def search_water_bill(params):
    return "账单金额:114元, 用水量:13吨, 是否欠费:否"

#查询停水公告
def search_water_notice(params):
    return "因香悦兰城12幢1单元检修停水，停水时间至12点，带来不便，敬请谅解！ 如有用水问题请致电公司24小时供水服务热线：0579-88927110。"

#查询收费标准
def query_fee_standards(params):
    return "1. 居民生活用水： 第一阶梯2.67元/吨（216吨） 第二阶梯3.47元/吨（120吨） 第三阶梯5.87元/吨。 合表居民用水2.77元/吨。\
            2. 农村居民生活用水（未接入城市污水管网）：  第一阶梯1.72元/吨（216吨） 第二阶梯2.52元/吨（120吨）  第三阶梯4.92元/吨。 合表居民用水1.82元/吨。\
            3. 非居民生活用水：非工业3.77元/吨，一般工业3.62元/吨，重污染工业4.52元/吨。 \
            4. 特种用水：4.67元/吨。 \
            5. 钱塘垅水厂（马涧、柏社区域水价）：生活用水：1.5元/吨，其他2.02元/吨。\
            6. 工业水厂水价：1.2元/吨，其中一般工业：2.6元/吨，重污染工业3.5元/吨，D类企业每吨加收0.8元/吨。"